package com.tyjt.common.redis;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.*;
import org.springframework.stereotype.Component;

import java.util.*;
import java.util.concurrent.TimeUnit;

/**
 * Redis 二进制数据操作工具类
 * 专门用于操作 byte[] 类型数据
 */
@Component
public class RedisByteUtil {

    @Autowired
    private RedisTemplate<String, byte[]> redisTemplate;

    // ============================== Key 操作 ==============================

    /**
     * 设置二进制缓存
     * @param key 键
     * @param value 二进制值
     */
    public void set(String key, byte[] value) {
        redisTemplate.opsForValue().set(key, value);
    }

    /**
     * 设置二进制缓存并设置过期时间
     * @param key 键
     * @param value 二进制值
     * @param timeout 过期时间
     * @param unit 时间单位
     */
    public void set(String key, byte[] value, long timeout, TimeUnit unit) {
        redisTemplate.opsForValue().set(key, value, timeout, unit);
    }

    /**
     * 获取二进制缓存
     * @param key 键
     * @return 二进制值
     */
    public byte[] get(String key) {
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 删除缓存
     * @param key 键
     * @return 是否删除成功
     */
    public Boolean delete(String key) {
        return redisTemplate.delete(key);
    }

    /**
     * 批量删除缓存
     * @param keys 键集合
     * @return 删除的数量
     */
    public Long delete(Collection<String> keys) {
        return redisTemplate.delete(keys);
    }

    /**
     * 设置过期时间
     * @param key 键
     * @param timeout 过期时间
     * @param unit 时间单位
     * @return 是否设置成功
     */
    public Boolean expire(String key, long timeout, TimeUnit unit) {
        return redisTemplate.expire(key, timeout, unit);
    }

    /**
     * 获取过期时间
     * @param key 键
     * @return 剩余过期时间(秒)
     */
    public Long getExpire(String key) {
        return redisTemplate.getExpire(key);
    }

    /**
     * 判断key是否存在
     * @param key 键
     * @return 是否存在
     */
    public Boolean hasKey(String key) {
        return redisTemplate.hasKey(key);
    }

    // ============================== Hash 操作 ==============================

    /**
     * 设置Hash二进制缓存
     * @param key 键
     * @param hashKey hash键
     * @param value 二进制值
     */
    public void hSet(String key, String hashKey, byte[] value) {
        redisTemplate.opsForHash().put(key, hashKey, value);
    }

    /**
     * 批量设置Hash二进制缓存
     * @param key 键
     * @param map 多个hash键值对
     */
    public void hSetAll(String key, Map<String, byte[]> map) {
        redisTemplate.opsForHash().putAll(key, map);
    }

    /**
     * 获取Hash二进制缓存
     * @param key 键
     * @param hashKey hash键
     * @return 二进制值
     */
    public byte[] hGet(String key, String hashKey) {
        return (byte[]) redisTemplate.opsForHash().get(key, hashKey);
    }

    /**
     * 获取所有Hash二进制键值对
     * @param key 键
     * @return 所有键值对
     */
    public Map<String, byte[]> hGetAll(String key) {
        Map<Object, Object> entries = redisTemplate.opsForHash().entries(key);
        Map<String, byte[]> result = new HashMap<>();
        entries.forEach((k, v) -> result.put((String) k, (byte[]) v));
        return result;
    }

    /**
     * 删除Hash缓存
     * @param key 键
     * @param hashKeys hash键集合
     * @return 删除的数量
     */
    public Long hDelete(String key, String... hashKeys) {
        return redisTemplate.opsForHash().delete(key, (Object[]) hashKeys);
    }

    /**
     * 判断Hash中是否存在某键
     * @param key 键
     * @param hashKey hash键
     * @return 是否存在
     */
    public Boolean hHasKey(String key, String hashKey) {
        return redisTemplate.opsForHash().hasKey(key, hashKey);
    }

    // ============================== List 操作 ==============================

    /**
     * 获取List二进制缓存
     * @param key 键
     * @param start 开始索引
     * @param end 结束索引
     * @return 列表片段
     */
    public List<byte[]> lGet(String key, long start, long end) {
        return redisTemplate.opsForList().range(key, start, end);
    }

    /**
     * 获取List长度
     * @param key 键
     * @return 列表长度
     */
    public Long lSize(String key) {
        return redisTemplate.opsForList().size(key);
    }

    /**
     * 通过索引获取List中的二进制值
     * @param key 键
     * @param index 索引
     * @return 二进制值
     */
    public byte[] lGetByIndex(String key, long index) {
        return redisTemplate.opsForList().index(key, index);
    }

    /**
     * 将二进制值放入List头部
     * @param key 键
     * @param value 二进制值
     * @return 操作后的列表长度
     */
    public Long lLeftPush(String key, byte[] value) {
        return redisTemplate.opsForList().leftPush(key, value);
    }

    /**
     * 将二进制值放入List尾部
     * @param key 键
     * @param value 二进制值
     * @return 操作后的列表长度
     */
    public Long lRightPush(String key, byte[] value) {
        return redisTemplate.opsForList().rightPush(key, value);
    }

    // ============================== Set 操作 ==============================

    /**
     * 向Set中添加二进制元素
     * @param key 键
     * @param values 二进制值集合
     * @return 成功添加的数量
     */
    public Long sAdd(String key, byte[]... values) {
        return redisTemplate.opsForSet().add(key, values);
    }

    /**
     * 获取Set中的所有二进制元素
     * @param key 键
     * @return 所有二进制元素
     */
    public Set<byte[]> sMembers(String key) {
        return redisTemplate.opsForSet().members(key);
    }

    /**
     * 判断Set中是否包含某二进制元素
     * @param key 键
     * @param value 二进制值
     * @return 是否包含
     */
    public Boolean sIsMember(String key, byte[] value) {
        return redisTemplate.opsForSet().isMember(key, value);
    }

    // ============================== 其他操作 ==============================

    /**
     * 获取Redis原生连接工厂
     * @return RedisConnectionFactory
     */
    public RedisConnectionFactory getConnectionFactory() {
        return redisTemplate.getConnectionFactory();
    }

    /**
     * 获取RedisTemplate实例
     * @return RedisTemplate
     */
    public RedisTemplate<String, byte[]> getRedisTemplate() {
        return redisTemplate;
    }

    /**
     * 获取ValueOperations实例
     * @return ValueOperations
     */
    public ValueOperations<String, byte[]> getValueOperations() {
        return redisTemplate.opsForValue();
    }

    /**
     * 获取HashOperations实例
     * @return HashOperations
     */
    public HashOperations<String, String, byte[]> getHashOperations() {
        return redisTemplate.opsForHash();
    }

    /**
     * 获取ListOperations实例
     * @return ListOperations
     */
    public ListOperations<String, byte[]> getListOperations() {
        return redisTemplate.opsForList();
    }

    /**
     * 获取SetOperations实例
     * @return SetOperations
     */
    public SetOperations<String, byte[]> getSetOperations() {
        return redisTemplate.opsForSet();
    }
}